package com.campus.counseling.service;

import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.dataset.DataSet;
import java.util.List;

public interface DL4JModelService {
    /**
     * 提取特征
     */
    INDArray extractFeatures(List<String> texts);
    
    /**
     * 预测
     */
    INDArray predict(INDArray features);

    /**
     * 加载模型
     */
    void loadModel();

    /**
     * 训练模型
     */
    void train(DataSet dataSet);
} 